/* This file is copied from JOS */
#ifndef _EKOS_X86_H_
#define _EKOS_X86_H_

#include <inc/types.h>

static __inline void breakpoint(void) __attribute__((always_inline));
static __inline uint8_t inb(int port) __attribute__((always_inline));
static __inline void insb(int port, void *addr, int cnt) __attribute__((always_inline));
static __inline uint16_t inw(int port) __attribute__((always_inline));
static __inline void insw(int port, void *addr, int cnt) __attribute__((always_inline));
static __inline uint32_t inl(int port) __attribute__((always_inline));
static __inline void insl(int port, void *addr, int cnt) __attribute__((always_inline));
static __inline void outb(int port, uint8_t data) __attribute__((always_inline));
static __inline void outsb(int port, const void *addr, int cnt) __attribute__((always_inline));
static __inline void outw(int port, uint16_t data) __attribute__((always_inline));
static __inline void outsw(int port, const void *addr, int cnt) __attribute__((always_inline));
static __inline void outsl(int port, const void *addr, int cnt) __attribute__((always_inline));
static __inline void outl(int port, uint32_t data) __attribute__((always_inline));
static __inline void invlpg(void *addr) __attribute__((always_inline));
static __inline void lidt(void *p) __attribute__((always_inline));
static __inline void lldt(uint16_t sel) __attribute__((always_inline));
static __inline void ltr(uint16_t sel) __attribute__((always_inline));
static __inline void lcr0(uint32_t val) __attribute__((always_inline));
static __inline uint32_t rcr0(void) __attribute__((always_inline));
static __inline uint32_t rcr2(void) __attribute__((always_inline));
static __inline void lcr3(uint32_t val) __attribute__((always_inline));
static __inline uint32_t rcr3(void) __attribute__((always_inline));
static __inline void lcr4(uint32_t val) __attribute__((always_inline));
static __inline uint32_t rcr4(void) __attribute__((always_inline));
static __inline void tlbflush(void) __attribute__((always_inline));
static __inline uint32_t read_eflags(void) __attribute__((always_inline));
static __inline void write_eflags(uint32_t eflags) __attribute__((always_inline));
static __inline uint32_t read_ebp(void) __attribute__((always_inline));
static __inline uint32_t read_esp(void) __attribute__((always_inline));
static __inline void cpuid(uint32_t info, uint32_t *eaxp, uint32_t *ebxp, uint32_t *ecxp, uint32_t *edxp);
static __inline uint64_t read_tsc(void) __attribute__((always_inline));
static __inline uint32_t xchg32(volatile uint32_t *addr, uint32_t newval) __attribute__((always_inline));
static __inline uint16_t xchg16(volatile uint16_t *addr, uint16_t newval) __attribute__((always_inline));
static __inline uint8_t xchg8(volatile uint8_t *addr, uint8_t newval) __attribute__((always_inline));

static __inline void
breakpoint(void)
{
     __asm __volatile("xchg %bx, %bx");
}

static __inline uint8_t
inb(int port)
{
     uint8_t data;
     __asm __volatile("inb %w1,%0" : "=a" (data) : "d" (port));
     return data;
}

static __inline void
insb(int port, void *addr, int cnt)
{
     __asm __volatile("cld\n\trepne\n\tinsb"			:
		      "=D" (addr), "=c" (cnt)		:
		      "d" (port), "0" (addr), "1" (cnt)	:
		      "memory", "cc");
}

static __inline uint16_t
inw(int port)
{
     uint16_t data;
     __asm __volatile("inw %w1,%0" : "=a" (data) : "d" (port));
     return data;
}

static __inline void
insw(int port, void *addr, int cnt)
{
     __asm __volatile("cld\n\trepne\n\tinsw"			:
		      "=D" (addr), "=c" (cnt)		:
		      "d" (port), "0" (addr), "1" (cnt)	:
		      "memory", "cc");
}

static __inline uint32_t
inl(int port)
{
     uint32_t data;
     __asm __volatile("inl %w1,%0" : "=a" (data) : "d" (port));
     return data;
}

static __inline void
insl(int port, void *addr, int cnt)
{
     __asm __volatile("cld\n\trepne\n\tinsl"			:
		      "=D" (addr), "=c" (cnt)		:
		      "d" (port), "0" (addr), "1" (cnt)	:
		      "memory", "cc");
}

static __inline void
outb(int port, uint8_t data)
{
     __asm __volatile("outb %0,%w1" : : "a" (data), "d" (port));
}

static __inline void
outsb(int port, const void *addr, int cnt)
{
     __asm __volatile("cld\n\trepne\n\toutsb"		:
		      "=S" (addr), "=c" (cnt)		:
		      "d" (port), "0" (addr), "1" (cnt)	:
		      "cc");
}

static __inline void
outw(int port, uint16_t data)
{
     __asm __volatile("outw %0,%w1" : : "a" (data), "d" (port));
}

static __inline void
outsw(int port, const void *addr, int cnt)
{
     __asm __volatile("cld\n\trepne\n\toutsw"		:
		      "=S" (addr), "=c" (cnt)		:
		      "d" (port), "0" (addr), "1" (cnt)	:
		      "cc");
}

static __inline void
outsl(int port, const void *addr, int cnt)
{
     __asm __volatile("cld\n\trepne\n\toutsl"		:
		      "=S" (addr), "=c" (cnt)		:
		      "d" (port), "0" (addr), "1" (cnt)	:
		      "cc");
}

static __inline void
outl(int port, uint32_t data)
{
     __asm __volatile("outl %0,%w1" : : "a" (data), "d" (port));
}

static __inline void 
invlpg(void *addr)
{ 
     __asm __volatile("invlpg (%0)" : : "r" (addr) : "memory");
}  

static __inline void
lidt(void *p)
{
     __asm __volatile("lidt (%0)" : : "r" (p));
}

static __inline void
lldt(uint16_t sel)
{
     __asm __volatile("lldt %0" : : "r" (sel));
}

static __inline void
ltr(uint16_t sel)
{
     __asm __volatile("ltr %0" : : "r" (sel));
}

static __inline void
lcr0(uint32_t val)
{
     __asm __volatile("movl %0,%%cr0" : : "r" (val));
}

static __inline uint32_t
rcr0(void)
{
     uint32_t val;
     __asm __volatile("movl %%cr0,%0" : "=r" (val));
     return val;
}

static __inline uint32_t
rcr2(void)
{
     uint32_t val;
     __asm __volatile("movl %%cr2,%0" : "=r" (val));
     return val;
}

static __inline void
lcr3(uint32_t val)
{
     __asm __volatile("movl %0,%%cr3" : : "r" (val));
}

static __inline uint32_t
rcr3(void)
{
     uint32_t val;
     __asm __volatile("movl %%cr3,%0" : "=r" (val));
     return val;
}

static __inline void
lcr4(uint32_t val)
{
     __asm __volatile("movl %0,%%cr4" : : "r" (val));
}

static __inline uint32_t
rcr4(void)
{
     uint32_t cr4;
     __asm __volatile("movl %%cr4,%0" : "=r" (cr4));
     return cr4;
}

static __inline void
tlbflush(void)
{
     uint32_t cr3;
     __asm __volatile("movl %%cr3,%0" : "=r" (cr3));
     __asm __volatile("movl %0,%%cr3" : : "r" (cr3));
}

static __inline uint32_t
read_eflags(void)
{
     uint32_t eflags;
     __asm __volatile("pushfl; popl %0" : "=r" (eflags));
     return eflags;
}

static __inline void
write_eflags(uint32_t eflags)
{
     __asm __volatile("pushl %0; popfl" : : "r" (eflags));
}

static __inline uint32_t
read_ebp(void)
{
     uint32_t ebp;
     __asm __volatile("movl %%ebp,%0" : "=r" (ebp));
     return ebp;
}

static __inline uint32_t
read_esp(void)
{
     uint32_t esp;
     __asm __volatile("movl %%esp,%0" : "=r" (esp));
     return esp;
}

static __inline void
cpuid(uint32_t info, uint32_t *eaxp, uint32_t *ebxp, uint32_t *ecxp, uint32_t *edxp)
{
     uint32_t eax, ebx, ecx, edx;
     asm volatile("cpuid" 
		  : "=a" (eax), "=b" (ebx), "=c" (ecx), "=d" (edx)
		  : "a" (info));
     if (eaxp)
	  *eaxp = eax;
     if (ebxp)
	  *ebxp = ebx;
     if (ecxp)
	  *ecxp = ecx;
     if (edxp)
	  *edxp = edx;
}

static __inline uint64_t
read_tsc(void)
{
     uint64_t tsc;
     __asm __volatile("rdtsc" : "=A" (tsc));
     return tsc;
}

static __inline uint32_t
xchg32(volatile uint32_t *addr, uint32_t newval)
{
     uint32_t result;
  
     // The + in "+m" denotes a read-modify-write operand.
     asm volatile("lock; xchgl %0, %1" :
		  "+m" (*addr), "=a" (result) :
		  "1" (newval) :
		  "cc");
     return result;
}

static __inline uint16_t
xchg16(volatile uint16_t *addr, uint16_t newval)
{
     uint16_t result;
  
     // The + in "+m" denotes a read-modify-write operand.
     asm volatile("lock; xchgw %0, %1" :
		  "+m" (*addr), "=a" (result) :
		  "1" (newval) :
		  "cc");
     return result;
}

static __inline uint8_t
xchg8(volatile uint8_t *addr, uint8_t newval)
{
     uint8_t result;
  
     // The + in "+m" denotes a read-modify-write operand.
     asm volatile("lock xchgb %0, %1" :
		  "+m" (*addr), "=a" (result) :
		  "1" (newval) :
		  "cc");
     return result;
}

#endif /* _EKOS_X86_H_ */
